# my_model_api_project/api_client.py

import requests
from utils.auth import get_api_key
from models.gpt import GPTModel
from models.bert import BertModel
# 导入其他模型类

class APIClient:
    def __init__(self, base_url, api_key):
        self.base_url = base_url
        self.api_key = api_key

    def call_gpt(self, prompt):
        model = GPTModel(self.base_url, self.api_key)
        return model.generate(prompt)

    def call_bert(self, text):
        model = BertModel(self.base_url, self.api_key)
        return model.analyze(text)

    # 其他API调用方法